package repository

import (
	"ai/internal/db/dao"
	"ai/internal/db/models"
)

type QuestionRepository struct {
	dao *dao.QuestionDAO
}

func NewQuestionRepository(questionDAO *dao.QuestionDAO) *QuestionRepository {
	return &QuestionRepository{dao: questionDAO}
}

// AddQuestion adds a new question to the database
func (repo *QuestionRepository) AddQuestion(question *models.Question) error {
	return repo.dao.AddQuestion(question)
}

// GetQuestionsByType retrieves questions of a specific type
func (repo *QuestionRepository) GetQuestionsByType(questionType string) ([]models.Question, error) {
	return repo.dao.GetQuestionsByType(questionType)
}

// ListQuestions retrieves all questions from the database
func (repo *QuestionRepository) ListQuestions() ([]models.Question, error) {
	return repo.dao.ListQuestions()
}

// Add methods for QuestionRepository as needed
